<?php

declare(strict_types=1);

namespace App\Test\Support\ElasticSearch;

use OpenSearch\Client;
use OpenSearch\ClientBuilder;
use OpenSearch\Endpoints\Bulk;
use OpenSearch\Endpoints\Index;
use OpenSearch\Serializers\SmartSerializer;
use Psr\Log\LoggerInterface;
use Shopware\Elasticsearch\Framework\ClientFactory;

/**
 * This factory ensures, that bulk and index steps are waiting to be completed on elasticsearch side.
 * This way the next request can expect to find the indexed documents in a search.
 *
 * @deprecated This is a bare copy with just a slight adjustment on the builder. Re-evaluate in case you update.
 */
final class WaitForBulkClientFactory extends ClientFactory
{
    /**
     * @param array{verify_server_cert: bool, cert_path?: string, cert_key_path?: string} $sslConfig
     */
    #[\Override]
    public static function createClient(string $hosts, LoggerInterface $logger, bool $debug, array $sslConfig): Client
    {
        $hosts = \array_filter(\explode(',', $hosts));

        $clientBuilder = ClientBuilder::create();
        $clientBuilder->setHosts($hosts);

        if ($debug) {
            $clientBuilder->setTracer($logger);
        }

        $clientBuilder->setLogger($logger);

        if ($sslConfig['verify_server_cert'] === false) {
            $clientBuilder->setSSLVerification(false);
        }

        if (isset($sslConfig['cert_path'])) {
            $clientBuilder->setSSLCert($sslConfig['cert_path'], $sslConfig['cert_password'] ?? null);
        }

        if (isset($sslConfig['cert_key_path'])) {
            $clientBuilder->setSSLKey($sslConfig['cert_key_path'], $sslConfig['cert_key_password'] ?? null);
        }

        // BEGIN CHANGE
        $clientBuilder->setEndpoint(function ($class) {
            $serializer = new SmartSerializer();

            if ($class === 'Bulk') {
                return new class($serializer) extends Bulk {
                    #[\Override]
                    public function setParams(array $params): void
                    {
                        // this is where the magic happens
                        $params['refresh'] ??= 'wait_for';

                        parent::setParams($params);
                    }
                };
            }

            if ($class === 'Index') {
                return new class() extends Index {
                    #[\Override]
                    public function setParams(array $params): void
                    {
                        // this is where the magic happens
                        $params['refresh'] ??= 'wait_for';

                        parent::setParams($params);
                    }
                };
            }

            $fullPath = '\\OpenSearch\\Endpoints\\' . $class;

            $reflection = new \ReflectionClass($fullPath);
            $constructor = $reflection->getConstructor();

            if ($constructor && $constructor->getParameters()) {
                return new $fullPath($serializer);
            } else {
                return new $fullPath();
            }
        });
        // END CHANGE

        return $clientBuilder->build();
    }
}